CUDA_VISIBLE_DEVICES=2,3,0,1
dataset='hmdb'
# modality=rgb
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/hmdb/single/rgb_trx/checkpoint30000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="train" --dataset=${dataset} --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=rgb
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/hmdb/single/rgb_trx/checkpoint30000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="test" --dataset=${dataset} --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=skeleton
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/kinetics/skeleton_trx/checkpoint50000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="train" --dataset=${dataset}\
# --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=skeleton
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/kinetics/skeleton_trx/checkpoint50000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="test" --dataset=${dataset} \
# --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

modality=flow
python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/hmdb/single/flow_trx/checkpoint40000.pt \
--path /media/star/zp/dataset/${dataset}/${modality}_l8 \
--mode="train" --dataset=${dataset} --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

modality=flow
python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/hmdb/single/flow_trx/checkpoint40000.pt \
--path /media/star/zp/dataset/${dataset}/${modality}_l8 \
--mode="test" --dataset=${dataset} --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=depth
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/hmdb/single/depth_trx/checkpoint30000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="train" --dataset=${dataset} --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=depth
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/hmdb/single/depth_trx/checkpoint30000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="test" --dataset=${dataset} --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=TG
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/kinetics/single/TG_trx/checkpoint25000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="train" --dataset=${dataset}\
# --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

# modality=TG
# python extract_feature.py --num_gpus=4 --modality=${modality} --checkpoint_dir=work/kinetics/single/TG_trx/checkpoint25000.pt \
# --path /media/star/zp/dataset/${dataset}/${modality}_l8 \
# --mode="test" --dataset=${dataset} \
# --traintestlist="imp_datasets/video_datasets/splits/hmdb51TrainTestlist"

